from unittest import TestCase
from mglearn import plots as mg_plots
from mglearn import datasets as mg_ds



class TestLinear(TestCase):
    def __init__(self, *args, **kwargs):
        super(TestLinear, self).__init__(*args, **kwargs)
        self.wave = mg_ds.make_wave(n_samples=60)


    def test_plot_wave_train(self):
        mg_plots.plot_linear_regression_wave()
        plot.show()


        